how to check Armstrong number in python